python train_mnist.py -c ./mnist.yaml